Add token input

:goat: