Add train routine & fix fl_ns