Update layout.py

fix pep8