Enable srun to submit job to multiple partition

Previously job would only run in first listed partition.