add token checking