Adding correctness tests