Adding first round of unit tests.