added screen shot methods to test