Token implementation + UTs