Fix up docs a bit, add TODO