Adds some tests
Merge pull request #1 from Achal-Aggarwal/unit-tests

Adds some tests