code: def normalize(X, mean=None, std=None):
      
      if mean is None or std is None:
          mean = X.mean(0)
          std = X.std(0)
      return (X - mean) / std
docs: Normalize X. If mean OR std is None, normalizes
      X to have mean 0 and std 1.