TODO: draw_layout.py