Simple test, fails local
Getting cuda stuff to work