Update TestGenerator.rb