Cleaned up imports in manage.py