Chamberlain Group Interview Question

add automation and unit test for the existing code