WIP implement train