Update main.rb

cleanup