a little update on main.rb