Finished out ability to take a screenshot
Gofmt it up