Added models for image