Clicker Game - Part 2

import pygame
import time
pygame.init()

from random import randint

'''create programme window'''

back = (200, 255, 255) #background color
mw = pygame.display.set_mode((500, 500)) #main window
mw.fill(back)
clock = pygame.time.Clock()

'''rectangle class'''

class Area():
    def __init__(self, x=0, y=0, width=10, height=10, color=None):
        self.rect = pygame.Rect(x, y, width, height) #rectangle
        self.fill_color = color

    def color(self, new_color):
        self.fill_color = new_color

    def fill(self):
        pygame.draw.rect(mw, self.fill_color, self.rect)

    def outline(self, frame_color, thickness): #outline of an existing rectangle
        pygame.draw.rect(mw, frame_color, self.rect, thickness)

    #---------------------------------------#
    def collidepoint(self, x, y):
        return self.rect.collidepoint(x,y)
    #---------------------------------------#

'''class label'''

class Label(Area):
    def set_text(self, text, fsize=12, text_color=(0, 0, 0)):
        self.image = pygame.font.SysFont('comic sans ms', fsize).render(text, True, text_color)

    def draw(self, shift_x=0, shift_y=0):
        self.fill()
        mw.blit(self.image, (self.rect.x + shift_x, self.rect.y + shift_y))

#---------------------------------------#
RED = (255, 0, 0)
GREEN = (0, 255, 51)
#---------------------------------------#

YELLOW = (255, 255, 0)
DARK_BLUE = (0, 0, 100)
BLUE = (80, 80, 255)

cards = []
num_cards = 4

x = 70

for i in range(num_cards):
    new_card = Label(x, 200, 70, 100, YELLOW) # x, y, width, height, colour
    new_card.outline(BLUE, 10)
    new_card.set_text('CLICK', 26)
    cards.append(new_card)
    x = x + 100 # Spacing between each card

#---------------------------------------#
wait=0    #ADD THIS!
#---------------------------------------#

while True:
    if wait == 0:
        wait = 20       # How long the word click will stay on-screen

        click = randint(1, num_cards)   # YOU HAVE THIS!
        for i in range(num_cards):
            cards[i].color(YELLOW)
            if (i + 1) == click:
                cards[i].draw(10, 40)
            else:
                cards[i].fill()

    else:               # ADD THIS ELSE!
        wait -= 1

    #checking the click on each tick:
    for event in pygame.event.get():
        if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
            x, y = event.pos
            for i in range(num_cards):
                if cards[i].collidepoint(x,y):
                    if i + 1 == click:
                        cards[i].color(GREEN)
                    else:
                        cards[i].color(RED)

                    cards[i].fill()

    pygame.display.update()
    clock.tick(40)