Energy calculation need to be checked carefully