Added comments 

Easier reading of code