Add testing

Fixes #16