Add missing tests

Closes #1