Enhance error highlighting (shown in red) to include warnings (shown in yellow).