Updated main.rb