Added in code for updateEnergy() method in O6_Model class