Changed the way correctness testing is done. It is still having issues