Split Connection class in connection.py file