Added collection generator for TestUnit