Removed linalg import that was unused