Fixed up formatting and added option for different weighting of disease selection