updates on mnist benchmark