cleanup, update TODO