[chore] Updating directive with optional ng-model