Fixing code blocks with headers immediatley after

Code blocks when drawn right before headings, didn't have any margin below them. I only added on line to help this out.