Making dataset 2 and 3 balance