Simple AI programming with pygame library