Add support for Add button