adding energy correction functions