Address some nits and update reference.