Update modified code