Added more tests.

Couple more tests.