Modify custom directive