Improve code snippet formatting