Screenshots work now with selenium driver