Added command line arg for directory