Change image migration to add category to images