putting tests in classes