modified layers.py