TODO: fix make_layout.py