Added potential energy in MPI comm. Having trouble with energy
conservation...