val boxModifier = Modifier
.fillMaxSize() or .fillMaxHeight()
.weight(1f)
while making sure that the box remains square, so basically, fill the maximum size/height as long as you are square? At the moment I have rectangles which is not looking great...