Adding numpy count_nonzero support