Implemented more unit tests, we now have 100% coverage.