fixed doc, and small mnist issues