works for NUM_FEATURES = 1