Removing deprecated YAML code
Create and use Work model and collection