Add tests

Problem/Motivation

This project does not have any automated tests. It should have some. I don't quite know where to start because I'm not sure what tests should be written for something like this (yet).

Remaining tasks

• Determine what kind of tests would be useful
• Write those tests