amending code to test on the hpc