Как использовать UIImagePickerController в Swift 4

Я пытаюсь получить изображение с камеры или фото-библиотеки и использую UIImageView для предварительного просмотра этой фотографии. Я использую приведенный ниже код для захвата изображения:

let picker = UIImagePickerController()
picker.delegate = self
picker.sourceType = .camera
present(picker, animated: true, completion: nil)

а затем я использую этот код, чтобы поместить изображение в imageView:

func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [String : Any]) {

    imageView.image = info[UIImagePickerControllerOriginalImage] as? UIImage
}

Я получаю разрешение, и приложение успешно загружается на моем iPhone но когда я делаю снимок, он не возвращается в основной вид, чтобы показать его в виде изображения. и он рушится. Кто-нибудь может мне помочь ?