Make tests match the output in README

I am sorry about the previous pull request, it doesn't have a matching test output for the test input which might confuse users. Here is a corrected version. 

Thanks