Updating models and including enrichment task class