minor update to TODO