udpated main.rb