check_n_threads function

Check if the input is a valid number of threads.