Added pygame testing