Generic directory

Change the setwd() to be generic