Improve formatting for code blocks