Main.rb updated