Updates to main.rb