Added take screenshot after query