Create layout.py