adding modified code