Pregunta de entrevista de OpenAI

Refactor the following simple code