import os, sys
import pygame
from pygame.locals import *

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