1. add predict function