Tests

**Is your feature request related to a problem? Please describe.**

We currently have no tests.

**Describe the solution you'd like**

Some tests.

**Describe alternatives you've considered**

Other tests?

**Additional context**

Tests are good.
