Adding unit tests.
7 files changed