Updated screenshot script