minor change to main.rb