Include a few unit tests