Some minor changes to compare_perplexity