Update test.rb

Code cleaned up