Add prediction hack and new model.