Tests

Fixes #7