Added a screenshot function