Align code blocks to cap height of headers

Credits to @vitorgalvao.