Added colouring
for different ks