Added my comments on your code