Reuse the same directive for add/edit component