tree class updated
updated test