Adding unit tests.
9 files changed