Added unit tests
adding unit tests