Added unit tests
Test suite complete