Modify script so it matches new directory