Starting to write proper unit tests.