Update to use yiistrap