Added colouring for different ks