Tree view is now working as a basic