feature selection added, not working yet