Added tree code from class.