LU Decomposition in Python