Add Correctness annotation and the rest of the correctness unit tests.