Changes to update_db