Updated slurm notebook