Updated stat collection and removed unused import