Added some class inheritance for setup and teardown.