Solving linear equations in python