Update main.rb

correction de bug