Add color to the prompt