Chamberlain Group interview question

add automation and unit test for the existing code