Minor changes to Main.rb