Fix bugs. Input desired # of features.