Update layout.py