fix the display using pygame.update