Add 3 line comment for using einsum