Fixed comments

fixed and finished comments on tree code