Added screenshot-at-end option.  fixes gh-17