Add test for Token class