Add a screenshot script