Michael | Tech Debt | Refactored the two test classes into one