Use flex instead of flex grow