Updated code working
Updated code working as per prof ip
Updated code working as per prof ip