Add a set of test code to main.rb--thanks @clamstew