modify main.rb