Adding setToken method. Closes #3