Comments and modifications on find_energies.py