Handle training stuff properly