import pygame
from pygame.locals import *
import sys
import time

def wait():
    time.sleep(0.0625)


pygame.init()
screen = pygame.display.set_mode((400, 300))
background = pygame.Surface((400, 300))

surface = pygame.Surface((10, 10))
surface.fill((0, 255, 0))

pos = [175, 125]

#Main loop, checking inputs 
while True:
    events = pygame.event.get()
    for event in events:
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()

        #X Axis movement    
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_RIGHT:
                pos[0] += 10
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                pos[0] -= 10

        #Y Axis movement
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP:
                wait()
                pos[1] -= 10
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_DOWN:
                wait()
                pos[1] += 10

        #Diagonal movement
        #if event.type == pygame.KEYDOWN:
        #   if event.key == pygame.K_LEFT and pygame.K_UP:
        #
        #Broken code.. can't find a solution to diagonals        
                

    #Constant Screen updates
    screen.blit(background, (0, 0))
    screen.blit(surface, pos)
    pygame.display.update()