added image model class