Added Tests for main.rb