Added comments & clarification.

Added some comments and cleanup.