small tweak to main.rb