tidy up plotting