Feature normalization works for the specific example but we need to extend it to any number of features and any number of samples.