Python script to play with the pygame library