Added screenshot functinality. But it now takes a screenshot of every test. Next fix will be to take a screenshot of only failing tests