Directive almost working