clean up Training a bit more