fix requirement for pygame and add simple test script