Removed unused imports from database.py