Added two simple tests