Python-сообщество

Здравствуйте!
Только начал изучать pygame и наткнулся на очень неприятную ошибку которую никак не могу решить.
буду очень благодарен если поможете
при выполнении после секунд 5 окно наглухо зависает

вот вам код

import pygame
pygame.init()

WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
GRAY = (125, 125, 125)
LIGHT_BLUE = (64, 128, 255)
GREEN = (0, 200, 64)
YELLOW = (225, 225, 0)
PINK = (230, 50, 230)

disp = pygame.display.set_mode((600,600))

xpos=300
ypos=300

w=10
h=10

done = False
while done == False:

for event in pygame.event.get():
if event.type == pygame.QUIT:
done = True

while xpos < 400:
pygame.time.wait(500)
pygame.draw.rect(disp,WHITE,(xpos,ypos,w,h))
pygame.display.update()
xpos += 1