Add simple tests to confirm correctness of implementations