Add simple visualization on mnist dataset