Updated database.py