not using DataFrames on train
improved perf