Updated to use the new screenshot method.
Merge pull request #2 from ChrisGrant/master

Updated to use the new screenshot method.