Update connect.py