Added a screenshot function

This is useful when used a screenshot feature on rspec suite, to grab
screenshots from failed tests.