#2 Add some simple tests