Updated the unit tests to match the spec in the readme