Add setup and teardown function. Use _TEST define for main.