add the token code to the model