Update directive to watch for child node changes