tweak comments, add TODO