Prepped code for saving images

Formatted button and prepped the code for saving the image.