Testing stuff with pygame