Modifiucations of energy routines to call polarisation functions