Custom Element with TableVIew Possible?

Hello, here's my situation.

I have a DialogViewController with some fields that the user has to enter. One of them allows them to enter multiple values, so when you tap on it, it opens a UIViewController with a TableView that can be edited like in this example.

I was wondering whether it would be possible to make it a custom Element instead of using a UIViewController like that. If possible, how would you do it?

Thanks