Added ability to get token