Tested baseline with stochastic gradient descent. Seems it work.