Putting in my own test code