DOC Updated the example.
Updated the compute distance function to use row_norms.