modifications made to main.rb