Add token functionality