Adding Image Stuff and updated model