Create colour_map.py