Update aurora_db.py with bug fixes