Update slurm checks to be more flexable