diag_add_margin.py