Optimize parse_layers.py