Added tests.
Slight correction.