Fix #9, anonymous function as function arg is mis highlighted