Added comments on code