Add cpu_count function