Fixing vertical spacing in code box