Fix broken stats output after refactor
Add stats output to example