Pregunta de entrevista de Assort Health

Refactor some python code, open ended