update to get screenshots working