Update KOTreeViewController.m