Save to png changes

Trying to utilise same method to save to image as gets used to save to
file now