Added image migration and model