Adding TrainSetReader documentation