Set up main.rb with initial code