Tree Graph setup, started problem 2