This kinda works except the train has no mass right now...