Fixed a problem in one of the ProblemSet constructors where a "train"
was used instead of a "test"