Trying to get directive to work