minimal improvement
fixed highlighting for variable-2 and variable-3