Initial pygame file

Shamelessly stole from memory puzzle and then removed unnecessary parts