Working on dataset balancing functionality