Added additional functionality for image production