Add token to User model