Added unit testing

added unit tests

added unit tests