Improve code box usability