Update connection_test.py