Added a custom directive