Pergunta de entrevista da empresa OpenAI

Refactor the following simple code