import pygame

def main():
    pygame.init()
    clock = pygame.time.Clock()

    #Create Screen
    screenSize = (500, 500)
    screen = pygame.display.set_mode((screenSize[0], screenSize[1]))
    pygame.display.set_caption("Paint")

    FPS = 60

    color = pygame.Color("Blue")

    pygame.display.update()

    while(True):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()

            if pygame.mouse.get_pressed()[0]:
                draw(screen, event.pos, color)

            keys = pygame.key.get_pressed()
            if keys[pygame.K_r]:
                color = pygame.Color("Red")
            elif keys[pygame.K_b]:
                color = pygame.Color("Blue")
            elif keys[pygame.K_e]:
                clear(screen)

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

def draw(screen, position, color):
    print(position)
    pygame.draw.circle(screen, color, position, 10, 0)

def clear(screen):
    screen.fill(pygame.Color("Black"))

main()