updated database.py