Added some simple tests