Merge pull request #15 from arpitgupta/master

fix of test code
Merge pull request #16 from arpitgupta/master

Use new test class