Update run_pm_dmo_NFW_fixed_timestep.md

            fig, ax = plt.subplots(1)

            ax.hist2d(p['pos'][:, 0], p['pos'][:, 1], bins=128)

            ax.set_aspect('equal')

            path = os.path.join(

                os.getenv('HW_BUILD', '.'), f'snap{ts.steps}_rank{mpi.rank}.png'

            )

            fig.savefig(path, bbox_inches='tight', dpi=200)

            fig.savefig(f'snap{ts.steps}_rank{mpi.rank}.png', bbox_inches='tight', dpi=200)

            plt.close(fig)

print('Simulation finished')

 No newline at end of file