* update nbs * rm unused functions * simplify loading * optionally return counts as well * update TODO * update TODO