Add functionality to read directory name from user