Added a unit test. All unit tests pass including the new one.