updating for NUM_FEATURES = 5