Move colouring functions to separate utils file