Update alpha_diversity.py