fixed some issues with selection, updated probabilities