如何实现两个垂直按钮在swift 5中删除

let button1 = UIButton(frame: CGRect(x: 50, y: 100, width: 100, height: 50))
button1.setTitle("Button 1", for: .normal)
button1.backgroundColor = UIColor.blue

let button2 = UIButton(frame: CGRect(x: 50, y: 200, width: 100, height: 50))
button2.setTitle("Button 2", for: .normal)
button2.backgroundColor = UIColor.red

button1.addTarget(self, action: #selector(deleteButton(_:)), for: .touchUpInside)
button2.addTarget(self, action: #selector(deleteButton(_:)), for: .touchUpInside)

@objc func deleteButton(_ sender: UIButton) {
    sender.removeFromSuperview()
}

self.view.addSubview(button1)
self.view.addSubview(button2)

import UIKit

class ViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        
        let button1 = UIButton(frame: CGRect(x: 50, y: 100, width: 100, height: 50))
        button1.setTitle("Button 1", for: .normal)
        button1.backgroundColor = UIColor.blue
        button1.addTarget(self, action: #selector(deleteButton(_:)), for: .touchUpInside)
        
        let button2 = UIButton(frame: CGRect(x: 50, y: 200, width: 100, height: 50))
        button2.setTitle("Button 2", for: .normal)
        button2.backgroundColor = UIColor.red
        button2.addTarget(self, action: #selector(deleteButton(_:)), for: .touchUpInside)
        
        self.view.addSubview(button1)
        self.view.addSubview(button2)
    }
    
    @objc func deleteButton(_ sender: UIButton) {
        sender.removeFromSuperview()
    }
}