Better error correction in code