trainMnist outputs the correct format. committing first submissions answers