Update connection.py