add feature of saving as PNG