Improved colour map and added function for getting colours from map