Add statistics collection