Adding a couple of tests.
Merge pull request #2 from gprasant/master

Added a couple of tests