Modified Code