works for NUM_FEATURES = 5