amend main.rb