Added controller for directive.