Improve code spacing