ChatGPT change this code to solve that tests:

> here, just have in mind it solves only test 1 and 2.

*Changes to cover more tests*

*Submits job interview tests*