Updated screenshot code per Brian