Add a few unit tests
Merge pull request #17 from jainanshul/test

Add a few unit tests