Refactored the code and made sure the tests still pass in a naive way.