add pygame.display.update()