Added untracked tree files
updated the code