Just added some comments on how to improve the solution.