Add tests for storing and retrieving token