Update energy.m