Correcting Bad Behavior with AI