Updated the unit tests to match the spec in the Readme