Add correction prompt