changes in Assignment4