Clean up syntax highlighting for draft