Update main.rb

bugfix