JavaFX Button Column Creation in Kotlin

createButton.kt

public fun <T> createButtonColumn(title: String, text: String, onClick: (T) -> Unit): TableColumn<T, T> {
    val buttonColumn = TableColumn<T, T>(title)

    buttonColumn.cellValueFactory = Callback { ReadOnlyObjectWrapper<T>(it.value) }

    buttonColumn.cellFactory = Callback {
        val button = Button()
        object : TableCell<T, T>() {
            override fun updateItem(item: T?, empty: Boolean) {
                super.updateItem(item, empty)
                if (item == null) {
                    graphic = null
                } else {
                    button.text = text
                    button.onAction = EventHandler<ActionEvent> {
                        onClick(item)
                    }
                    graphic = button
                }
            }
        }
    }
    return buttonColumn
}