Add specs for training set builder