improve demo of directive