Add callback function that allows you to give a colour given a string

[vpineda1996]

I saw the other commit but I feel like this might be better since now we can customize how we select colours :D