don't pass around train objects all over the place