added additional comment as per the insturctions given in the course