trying to tidy up, also added the colour maps