linear algebra using python