python pygame mainloop