Update TODO in comments