Changed test.rb code