Started adding unit tests.