Fixed test cases
Merge pull request #7 from riker09/fix-test-cases

Fixed test cases