Add tests

:warning: **WIP** :warning: 

/cc #7