Better unit tests.

#2