Updated highlighting for equal sign in function expressions