Added feature selection plot