Update main.rb

L6 working