Correction to handle multiple test types.
Correction to handle multiple test types.
Merge pull request #180 from roydude/v1-train

Correction to handle multiple test types.