Added pygame test code