Added screenshot function.