add AI correction