adding feature selection stuff