Add in WillUpdate and DidUpdate hooks.