fix typos and update TODO