Fixing colors in prompt