matrix multiplication in python user input