work calculation and energy graphs in driven systems