Add code to draw the pinball domain with pygame