Added TODO item
Minor doc fix