Added Tree class (forgot to add earlier sorry)