Moved Connection to connection.py