Stripping in-class work out of main.rb
Update assignment