Add tests

:warning: **WIP** :warning: 

Fixes #7.