Added screenshot ability to customer tests