Update main.rb

A bit of syntax sugar