Add caching to diversity functions

Right now the diversity functions in anopheles.py are not saved to a cache. Unless there is a specific reason why this isn't the case, it would be great (and I think quite straightforward?) to add this functionality. Specifically ```roh_hmm``` and ```cohort_diversity_stats```.