Oxolo interview question

Improve the code of an open-source algorithm