Process diagram with pygame