Improve line wrapping in code