Added distribution-choosing parameter for dist.py