Small changes to tree