Add token unit test