solved bug on dataset selection
Improved interface