Update database.py