Changes to main.rb