SharedElementsRoot {
    Surface(
        modifier = Modifier
            .fillMaxSize(),
        color = MaterialTheme.colors.background
    ) {
        val navController = rememberNavController()
        NavHost(
            navController = navController,
            startDestination = "First_screen"
        ) {
            composable(route = "First_screen") {
                LazyColumn(
                    contentPadding = PaddingValues(vertical = 100.dp) // Simulating position in the list
                ) {
                    item {
                        SharedElement(
                            key = "Container",
                            screenKey = "First"
                        ) {
                            SharedCard {
                                navController.navigate("Second_screen")
                            }
                        }
                    }
                }
            }
            composable(route = "Second_screen") {
                SharedElement(
                    key = "Container",
                    screenKey = "Second"
                ) {
                    SharedCardContent {
                        navController.popBackStack()
                    }
                }
            }
        }
    }
}

@Composable
fun SharedCard(
    onClick: () -> Unit
) {
    Card(
        modifier = Modifier
            .fillMaxWidth(),
        elevation = 2.dp,
        shape = RoundedCornerShape(5.dp),
        onClick = onClick
    ) {
        Column(
            modifier = Modifier
                .padding(
                    top = 10.dp,
                    bottom = 8.dp,
                    start = 15.dp,
                    end = 15.dp
                )
        ) {
            Text(
                text = "Card title",
                style = MaterialTheme.typography.h6
            )

            Spacer(modifier = Modifier.height(10.dp))

            Column(
                verticalArrangement = Arrangement.spacedBy(8.dp)
            ) {

                SharedElement(
                    key = "author",
                    screenKey = "First"
                ) {
                    Text("Author name")
                }

                SharedElement(
                    key = "subject",
                    screenKey = "First"
                ) {
                    Text("Subject name")
                }
            }
        }
    }
}

@Composable
fun SharedCardContent(
    onClick: () -> Unit
) {
    Surface(
        modifier = Modifier
            .fillMaxWidth()
            .clickable(onClick = onClick),
        color = MaterialTheme.colors.surface,
        elevation = 1.dp
    ) {
        Column(
            modifier = Modifier
                .fillMaxWidth()
                .padding(20.dp),
            verticalArrangement = Arrangement.spacedBy(10.dp)
        ) {
            SharedElement(
                key = "author",
                screenKey = "Second"
            ) {
                Text("Author: Author name")
            }

            SharedElement(
                key = "subject",
                screenKey = "Second"
            ) {
                Text("Subject: Subject name")
            }

        }
    }
}

    fun onElementDisposed(elementInfo: SharedElementInfo) {
        activeAnimations.put(elementInfo.key, elementInfo.screenKey)
        choreographer.postCallback(elementInfo) {
            val tracker = getTracker(elementInfo)
            tracker.onElementUnregistered(elementInfo)
            if (tracker.isEmpty) trackers.remove(elementInfo.key)
        }
    }

@Composable
private fun SharedElementTransitionsOverlay(rootState: SharedElementsRootState) {
  // unchanged...
                scope.launch {
                   // unchanged...
                    activeScreenKey = rootState.activeAnimations.get(startElement.info.key)
                    val correctTarget = activeScreenKey == startElement.info.screenKey
                    Timber.tag("wtf").i("firing onTransitionFinished() for startElement ${startElement.info.screenKey}. Is this correct? $correctTarget")
                    transition.onTransitionFinished()
                }
}