added dataset (not working with RF)