Done for energy calculation. Remove TODO.