the code works and the tests pass, can you make the code easier to maintain? - View it on GitHub