Added register updates. Minor nits.