add test to main.rb to test