Add tests

Only parts of the codebase is currently covered by tests. Adding new tests would be valuable to:
- Catch bugs early, especially as multiple people start working on the code
- Help the documentation effort since good tests are small chunks of code that illustrate specific functionalities in action