https://cscprogrammingtutorials.com/program-to-calculate-energy/