Added a setup section to the tests