latex.lang: improve math highlighting