Refactoring custom HTML to custom Directive.

-Dx