Add token generation