【拼图游戏】麻烦各位大神看看以下拼图游戏代码有什么错误可以改进的地方，谢谢啦

import pygame
import random

# 初始化pygame
pygame.init()

# 设置窗口大小和标题
width, height = 300, 300
screen = pygame.display.set_mode((width, height))
pygame.display.set_caption("拼图游戏")

# 定义拼图块的大小和数量
block_size = 100
rows, cols = 3, 3

# 加载拼图块的图片
image = pygame.image.load("C:/Users/ASUS/pintu/pintu/images/yuantu.jpg")
image = pygame.transform.scale(image, (block_size*cols, block_size*rows)) #将图片按拼图像素调整大小

# 创建拼图块的列表
blocks = []
for i in range(rows):
    for j in range(cols):
        rect = pygame.Rect(j * block_size, i * block_size, block_size, block_size)
        block = image.subsurface(rect)
        blocks.append(block)

# 打乱拼图块的顺序
random.shuffle(blocks)

# 创建拼图块的初始位置列表
positions = [(j * block_size, i * block_size) for i in range(rows) for j in range(cols)]
mouse_click_time = 0
click_first = -1
# 记录鼠标第二次点击的图片
click_second = -1
# 游戏主循环
running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        elif event.type == pygame.MOUSEBUTTONDOWN:
            # 获取鼠标点击的位置
            mouse_pos = pygame.mouse.get_pos()
            
            # 找到被点击的拼图块
            clicked_block = None
            for i, position in enumerate(positions):
                if pygame.Rect(position, (block_size, block_size)).collidepoint(mouse_pos):
                
                    if mouse_click_time % 2 == 0:
                        clicked_block = i
                    elif mouse_click_time % 2 == 1:
                        # 交换两个图片的位置
                        positions[clicked_block], positions[i] = positions[i], positions[clicked_block]    
                    mouse_click_time =mouse_click_time +1
                
    
    # 绘制拼图块
    for i, position in enumerate(positions):
        screen.blit(blocks[i], position)
    
    # 更新屏幕
    pygame.display.flip()

# 退出游戏
pygame.quit()