Add setup and teardown hooks