[ios - swift] textfield return

override func viewDidLoad() {
        super.viewDidLoad()
        
        emailTextField.delegate = self
        pwTextField.delegate = self
  }

extension LoginView: UITextFieldDelegate {
  func textFieldShouldReturn(_ textField: UITextField) -> Bool {
    if textField == emailTextField {
      pwTextField.becomeFirstResponder()
    } else {
      pw.resignFirstResponder()
    }
    return true
  }
}