working on adding tests.