Added a unit test suite