moving stuff around
test args for matmul
missing imports
starting to use base for test_probe