Tree is now a function that takes a directory argument