Fixing highlighting issues inside of function body