Adding Updated code