Updated TODO in readme