Minor clean-up plus update TODO(s)