Improve colouring