Added unit tests to add