Adding ImageModel and imageSchema