example program for pygame