Made changes to test_connection