Better testing with setup/teardown