Reverting changes to database.py