Is this extension function from the docs correct?

# compose

[Mehdi Haghgoo]

Is this extension function from the docs correct?

[Mehdi Haghgoo]

enum class WindowSizeClass { COMPACT, MEDIUM, EXPANDED }

@Composable
fun Activity.rememberWindowSizeClass() {
    val configuration = LocalConfiguration.current
    val windowMetrics = remember(configuration) {
        WindowMetricsCalculator.getOrCreate()
            .computeCurrentWindowMetrics(this)
    }
    val windowDpSize = with(LocalDensity.current) {
        windowMetrics.bounds.toComposeRect().size.toDpSize()
    }
    val widthWindowSizeClass = when {
        windowDpSize.width < 600.dp -> WindowSizeClass.COMPACT
        windowDpSize.width < 840.dp -> WindowSizeClass.MEDIUM
        else -> WindowSizeClass.EXPANDED
    }

val heightWindowSizeClass = when {
        windowDpSize.height < 480.dp -> WindowSizeClass.COMPACT
        windowDpSize.height < 900.dp -> WindowSizeClass.MEDIUM
        else -> WindowSizeClass.EXPANDED
    }

// Use widthWindowSizeClass and heightWindowSizeClass
}

[Mehdi Haghgoo]

İt doesn't return anything.

[Mehdi Haghgoo]

Shouldn't it return a window class?
Doc page: https://developer.android.com/guide/topics/large-screens/support-different-screen-sizes#compose