[WIP] Unit tests

Resolves #7.