Domanda di colloquio di OpenAI

Refactor the following simple code