Update spiral_transmission.py