Add tests for ml model creation, patch ml creation