Updating to work with new version of yii2