Tweaks, fix pause

examples/falling_balls_with_web/src/commonMain/kotlin/fallingBalls/Game.kt

@@ -42,17 +42,17 @@ abstract class Game {
 
     var numBlocks by mutableStateOf(5)
 
-    fun isInBoundaries(pieceData: PieceData): Boolean = pieceData.position < size.height
+    fun isInBoundaries(pieceData: PieceData) = pieceData.position < size.height
 
-    abstract fun saveTime()
+    abstract fun now(): Long
 
     fun togglePause() {
         paused = !paused
-        saveTime()
+        previousTime = Long.MAX_VALUE
     }
 
     fun start() {
-        saveTime()
+        previousTime = now()
         startTime = previousTime
         clicked = 0
         started = true

examples/falling_balls_with_web/src/commonMain/kotlin/fallingBalls/Piece.kt

@@ -16,7 +16,7 @@ import jetbrains.compose.common.shapes.CircleShape
 import org.jetbrains.compose.common.demo.position
 
 @Composable
-fun piece(index: Int, piece: PieceData) {
+fun Piece(index: Int, piece: PieceData) {
     val boxSize = 40.dp
     Box(
         Modifier

examples/falling_balls_with_web/src/commonMain/kotlin/fallingBalls/fallingBalls.kt

@@ -86,7 +86,7 @@ fun fallingBalls(game: Game) {
                     }
             ) {
                 game.pieces.forEachIndexed { index, piece ->
-                    piece(index, piece)
+                    Piece(index, piece)
                 }
             }
         }

examples/falling_balls_with_web/src/jsMain/kotlin/androidx/compose/web/with-web/App.kt

@@ -11,25 +11,20 @@ import androidx.compose.web.css.Style
 import org.jetbrains.compose.web.ui.Styles
 
 class JsGame : Game() {
-    override fun saveTime() {
+    override fun now() = window.performance.now().toLong()
-        previousTime = window.performance.now().toLong()
-    }
 }
 
 fun main() {
     val root = document.getElementById("root") as HTMLElement
 
-    renderComposable(
+    renderComposable(root = root) {
-        root = root
-    ) {
         Style(Styles)
-        fallingBalls(
+        val game = remember {
-            remember {
+            JsGame().apply {
-                JsGame()?.apply {
+              width = root.offsetWidth
-                    width = root.offsetWidth
+              height = root.offsetHeight
-                    height = root.offsetHeight
-                }
             }
-        )
+        }
+        fallingBalls(game)
     }
 }

examples/falling_balls_with_web/src/jvmMain/kotlin/App.kt

@@ -7,16 +7,14 @@ import org.jetbrains.compose.demo.falling.Game
 import androidx.compose.runtime.remember
 
 class JvmGame : Game() {
-    override fun saveTime() {
+    override fun now() = System.nanoTime()
-        previousTime = System.nanoTime()
-    }
 }
 
 fun main() {
     Window(title = "Demo", size = IntSize(600, 400)) {
         fallingBalls(
             remember {
-                JvmGame()?.apply {
+                JvmGame().apply {
                     width = 600
                     height = 400
                 }