Is the idea to just save a profile plot after the simulation is completed? If so, please consider using after_main_loop instead of after_step.