adding correctness test