Added some comments on how to improve code