Adds token functionality to test client