Added feature to split dataset according to a fixed list