energyPrediction working