优化一下code block