Updated trees to homework 4