Added setup and teardown functions for tests