Fix syntax highlighting within class attribute and function parameter initializers

Hi, i've made few aditional fixes.

It fixes use cases like this:

```typescript
class MyClass {
  attr = initfn('string arg');
  someMethod(arg: number[] = [1,2,3]) { }
}
```