Add unit test suite