Add get_token for getting input

That's probably a horrible name for it but whatever.