Added more code to the ImageEnhancer