Refactore token validation