Adding aconnection to DB in model/__init__.py file