[MPI] Add tokens to the statistics
Created by: inailuig
Adds (optional) tokens to all relevant functions in nk.stats.
Let me know if I should change the last two commits to not return the token to avoid breaking compatibility of possible external users of nk.stats.statistics for now