Added connect to models.py