updated tests
Merge pull request #5 from vikram7/tests

updated tests