Add ability to save generated image