Improved  the directory walker class