We adjusted the calculate_free_energy method to consider the axis limits specified by the --xlim_inf, --xlim_sup, --ylim_inf, and --ylim_sup arguments during the contour data calculation phase. Now, the grid on which the probability density and free energy are calculated is created based on the specified limits, ensuring that contour data are generated for the entire desired visualization area.
Furthermore, a check was implemented to ensure that the specified limits are respected when plotting the free energy landscape, adjusting the plot_energy_landscape and plot_3D_energy_landscape methods to consistently apply these limits.