added feature selection