added some training stuff