Adding code for creation of images