Update discovery.py