Neaten main.rb code