Exposes token class to controller.

Fixes #5