Your suggestion on how to improve the solution above​