Add prox newton and predict functions