Added unit tests
Merge pull request #2 from karthikbgl/master

Added unit tests