update with token
Updated the tests