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

Unit tests