Changed dataset splitting to take label proportions into account, and changed split to be 50-50.