def normalize(X, mean, std):
    """Normalization of array
    Args:
       X (np.array): Dataset of shape (N, D)
       mean (np.array): Mean of shape (D, )
       std (float): Standard deviation of shape(D, )
    """
    return (X - mean) / std