Updating: some testing code for matmul.