Example of Linear algebra in python