Added notes in code