Помогите добавить в код на Python частицы

import pygame
import random
from pygame.locals import *

# Инициализация Pygame
pygame.init()

# Размеры экрана
screen_width = 800
screen_height = 600

# Цвета
BLACK = (0, 0, 0)

# Создание экрана
screen = pygame.display.set_mode((screen_width, screen_height))
pygame.display.set_caption("Particle Effects")

# Создание частиц
class Particle(pygame.sprite.Sprite):
    def __init__(self, pos, speed):
        super().__init__()
        self.image = pygame.Surface((4, 4))
        self.image.fill((255, 255, 255))
        self.rect = self.image.get_rect(center=pos)
        self.speed = speed

    def update(self):
        self.rect.x += self.speed[0]
        self.rect.y += self.speed[1]
        if self.rect.left > screen_width or self.rect.right < 0 or self.rect.top > screen_height or self.rect.bottom < 0:
            self.kill()

# Создание группы для частиц
all_particles = pygame.sprite.Group()

# Основной цикл
clock = pygame.time.Clock()
running = True
while running:
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False

    # Генерация новых частиц
    particle_pos = (random.randint(0, screen_width), screen_height)
    particle_speed = (random.uniform(-2, 2), -random.uniform(1, 4))
    particle = Particle(particle_pos, particle_speed)
    all_particles.add(particle)

    # Обновление частиц
    all_particles.update()

    # Очистка экрана
    screen.fill(BLACK)

    # Отрисовка частиц
    all_particles.draw(screen)

    pygame.display.flip()
    clock.tick(60)

pygame.quit()