update TODO and remove unused imports