Add date variables to predict function