Add Tree tests