Added get_distribution