added a setup and teardown function to the unit test