Very rudimentary start to adding tests