[NB] Add predict method.  Not sure it's correct.
[NB] Use X if Y syntax
[NB] Store total counts all along the way