Added few simple tests