WIP: Add tests.