modify update_db function