Add context to BSTree traverse