add a routine for strain energy