Update assignment1_main.rb