added simple tests!