more support for multiple training sets 2