Add top/bottom margin to code