Add unit tests.

Ack.