Thinking about setting up tests for some parts at least

It's very tedious to write all these out though