Fixed screenshot code