I think I have the train object working