Make tests match the output in README
Merge pull request #7 from minhajuddin/master

Make tests match the output in README