Updated TreeMenuFragment, left comments for Chris