added code to save as png
init