Fix TOC and code coloring in docs