Updated test code

Update from announcement on T-Square