Code updates for AngularJS including directives