Fixing train to pass info to train controller