user define matrix multiplication in python