update for new DOM