improve code block