adding predict endpoint