Adding Train Class