Altered the distribution handling

Choosing the distributions a simple user input.