Update code_challenge.rb