Adding Tree class