Add functional tests

Problem/Motivation

We should have tests for the general functionality and tokens to ensure everything works as expected.

Steps to reproduce

Proposed resolution

Remaining tasks

User interface changes

API changes

Data model changes