clean up matmul example