Improve code-block styles on small screens