small change in database.py