Refactor connection objects into connection.py

Update the tests and objects to point to the new place