Extended the code styling to the energy computation tool