Changes to dataset .. classification