rejigged tests and added screenshot takerer