improve count module