Adding new changes as per the feedback on pull request.