Updated directive to work with NgModelController