added uncommented pygame code