Migrage code to use the classification bundle
update migration process