Add wrapper for action buttons