Add auto highlight code support