Added a couple of simple tests