Adding momentum SGD.  Also better reporting of statistics.