fixed feature selection
feature selection fixed now