added comments - code cleanup