1/. add new predict function