adding support for other custom directives