Add testing

Closes #3