Don't return trained model in parallel
Cleanup