improve AI code output