fikser litt på stochastic gradient. funker ikke helt enda