Updated TODO section