Add error correction