Added unit tests

Added unit tests to test project