trying to make plotting more standardised