Refactored and added tests to add 100% coverage.