Hi there! I just noticed that the ability to sample from selected diseases is not present anymore: https://github.com/foundation29org/RareCrowds/blob/832bb6f7b61cc9c54da0b0100d798ea790b4cf2b/rarecrowds/utils/patient_sim.py#L124 It would be highly appreciated ;)