Adding in example and fixing tests.