Add support for the universal edit button