pygame button that change screen code example

Example: how to update screen in pygame

pygame.display.update()