Fixed tests.  Can now pass a function in between url load and screenshot