Mostly complete highlights for code mode

[Slickytail]

Note that these queries will throw errors using the current treesitter parser. I've made a bunch of fixes to the parser which go along with these queries, which I'll submit as PR to the parser repo.