make predict running faster