Add code snippets' handling to the theme