more consistent variable naming in train.jl