game 2D write in python with pygame library