updated the notebook

added a new plot and code to the feature selection section