Change initialization back, comment out modelgradient code