Complete - task : Modified some code