Small changes

Trying to solve unit training as a created unit problem