Update TestGenerator.rb

Different Tasks