Update rbtree code