Write more tests

## Description
This repo barely has any tests. Some classes are well tested while others have zero tests whatsoever. In particular, more unit tests would be welcome. For reference, our test coverage is only 26%.