Plotting energy from gromacs trajectories