Added colouring function