Added correctness test