Update lab2.rb