Test code running on GPU
TODO list updated