Added color fix to prompt code