Work on a few functs
Trying out pygame.