Making the modify option work properly.