def pro(A,B):
        """
        Compute the trace of the matrix product between A and B.
        
        Arguments:
            A,B -- square matrices 
        
        Return:
            com -- Scalar
        """
        prod = (A.dot(B)).trace()
        return prod