add TODO, other tweaks