Adding train and test for labels randomised balanced.