Finally got the energy code working!!! TODO: Add functionality xD