The documentation of the function FreeEnergyAveraged states that the parameter n_grid is [int; 3], but in the code [int; 2] is used