grad descent seems to be working