Adding custom directive for elemental template