Calculation of elastic potential energy with different masses