Fixed tests from failing
Fixed assertion failing
Merge pull request #7 from sunnyratilal/unit-tests

Unit tests