added comments

Just to my code because I was unsatisfied.