[PyGame] Apparently module has no attribute display?

[Klar]

Hey, it's me again! I have no experience with PyGame at all!
Anyway, I'm importing this code:

import pygame

screen = pygame.display.set_mode((640, 400))

running = 1

while running:
	event = pygame.event.poll()
	if event.type == pygame.quit:
		running = 0

and it returns

Error:
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\[path]\pygametesteroo.py", line 3, in <module>
    screen = pygame.display.set_mode((640, 400))
AttributeError: 'module' object has no attribute 'display'