TODO minor updates