Quick updates to TODO