Initial draft for unit test.

Fixes #4