adding code to main in driver to create directory iterator an get files