WIP: add tests.