added some changes
Added comment