Add custom directives support