Update assignment.rb