I have modified the get_colour function to something that I think is nicer. It is slightly buggy and doesn't display colours correctly yet- I shall look into this tomorrow.