minor cleanup in main.rb