Add TODO and fix README