fix feature selection bug. demo.sh