Simplify gemm example