Update matrix_tools.py