added test to main.rb