Added user input for energy calcs