Remove stale TODOs and imports