Update code for Yii2