Forgot to add changes in db.py to last commit.