Add button as a directive