Make simple trainer.

Вынеси из if, а то дубликация