Add token module as dependency