Improve how code snippets are displayed