Create energy plot

plotf.py

@@ -31,6 +31,9 @@ def plots(file, settings_passed):
     
     # Trajectories
     trajectory(bodies, settings_passed, inds=settings_passed.plot.inds)
+    # Energies
+    plot_energy(bodies, settings_passed)
+    
 #     plot_eccentricity(semi_major_axis, e)
 
 
@@ -140,6 +143,7 @@ def plot_energy(bodies, settings):
     plt.xlabel('time $[y_J]$', fontsize = settings.plot.fontsize_title)
     plt.ylabel('Energy', fontsize = settings.plot.fontsize_title)
     plt.title('Energy')
+    plt.legend(loc='upper right', fontsize = settings.plot.fontsize_legend)
     plt.show()