Unit tests

Complete test coverage for current code base.