Updated TODO section of readme