Made small changes to recognize the change.