hmm or model1 in same file... gradient works in both models now