Added test file for pygame