CodeChallenge minor change