Improve the linear algebra example