* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
* Test code clean up
thrust variable is currently a 1D numpy array so the matmul function throws an error when executed, converting thrust to a scalar before the matmul operation solves the issue.
