Cleanup / fixes on the large scale text classification tutorial