Change to use ng-model