Prompt color based on return code