Improve code display