updated relay.py