added changes to main.rb