homework code  added to the code.rb file