Made the colour code neater