added extra unit tests.