Added lines to test.rb