TODO: fix draw_layout.py