Update layers.py