Training set behaves as expected