added directive with link function