Improved model training / testing setup to work more automated (SLURM bash script now get generated automatically)